package com.chilly.vidio.study.leetcode;

import java.util.Arrays;

/**
 * 80. 删除排序数组中的重复项 II
 * 给定一个增序排列数组 nums ，你需要在 原地 删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。
 * <p>
 * 不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
 * <p>
 * 示例 1：
 * 输入：nums = [1,1,1,2,2,3]
 * 输出：5, nums = [1,1,2,2,3]
 * 解释：函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。
 * 你不需要考虑数组中超出新长度后面的元素。
 *
 * @author Chilly Cui
 * @createDate 2021/2/27
 */
public class No26RemoveDuplicatedFromSortedArray2No80 {

    public static void main(String[] args) {
        Solution solution = new Solution();
        int[] nums = {1, 1, 1, 2, 2, 3};
//        expect：5, nums = [1,1,2,2,3]
        System.out.println(solution.removeDuplicates0(nums));
        System.out.println(Arrays.toString(nums));
//        [1, 1, 2, 2, 3, 3]
    }

    static class Solution {

        //优化点：要求每个元素最多出现k次
        public int removeDuplicates(int[] nums, int k) {
            if (nums == null) {
                return 0;
            }
            if (nums.length <= k) {
                return nums.length;
            }
            // 1.定义[0,index] 是修改后的满足要求的数组区间,这里已经把0 1 2 ...k- 1 ,共k个数 放进去了
            int index = k - 1;
            // 2.判断终止条件
            for (int i = k; i < nums.length; i++) {
                // 3.指针移动条件
                if (nums[i] != nums[index - k + 1]) {
                    index++;
                    nums[index] = nums[i];
                }
            }
            // 4.判断返回值
            return index + 1;
        }

        //每个元素最多出现两次
        public int removeDuplicates2(int[] nums) {
            // 1.定义[0,k)表示修改后的数组 即不重复的元素
            int k = 0;
            // 2.终止条件
            for (int i = 0; i < nums.length; i++) {
                // 3.指针运动条件
                if (k == 0 || nums[k - 1] != nums[i]) {
                    nums[k] = nums[i];
                    k++;
                }
            }
            // 4.根据定义确定返回值
            return k;
        }

        // error 自己写的
        public int removeDuplicates(int[] nums) {
            int k = 2;
            for (int i = 2; i < nums.length; i++) {
                if (nums[i] != nums[k-2]) {
                    nums[k] = nums[i];
                    k++;
                } else {
                    int re = i - k + 1;
                    if (re > 2) {
                        k = i;
                        nums[k] = nums[i];
                    }
                }
            }
            return 0;
        }

        //每个元素最多出现两次
        public int removeDuplicates0(int[] nums) {
            if (nums.length <= 2) {
                return nums.length;
            }

            int index = 2;
            for (int i = 2; i < nums.length; i++) {
                if (nums[i] != nums[index - 2]) {
                    nums[index++] = nums[i];
                }
            }

            return index;
        }
    }
}
